package com.wangshupo.app.webSocket;

import javax.servlet.http.HttpSession;
import javax.websocket.HandshakeResponse;
import javax.websocket.server.HandshakeRequest;
import javax.websocket.server.ServerEndpointConfig;

public class GetHttpSessionConfig extends ServerEndpointConfig.Configurator {
    /**
     * 获取HttpSession 并保存到 ServerEndpointConfig 对象中，后续通过 EndpointConfig获 取存入的数据
     */
    @Override
    public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {

        // 获取 HttpSession 对象
        HttpSession httpSession = (HttpSession) request.getHttpSession();
        // 在 @ServiceEndpoint 类的 onOpen 方法就能通过 EndpointConfig 对象获 取在这里存入的数据  TODO 存入登录用户的HttpSession
        sec.getUserProperties().put(HttpSession.class.getName(), httpSession);
    }



}
